package mylab.starters.dag;

import com.google.common.collect.Maps;

import java.util.Map;

import lombok.NonNull;
import lombok.experimental.UtilityClass;

/**
 * 获取DagContext上下文的工具类
 */
@UtilityClass
@SuppressWarnings("all")
public class DagContextHolder {
    Map<String, DagContext> holder = Maps.newConcurrentMap();

    public void set(@NonNull DagContext dagContext) {
        holder.put(dagContext.getGraph().getId(), dagContext);
    }

    public DagContext get(@NonNull String graphId) {
        return holder.get(graphId);
    }

    public void remove(@NonNull String graphId) {
        holder.remove(graphId);
    }

}